require_relative '../../../spec_helper'
require_relative 'fixtures/classes'

describe "Enumerator::Lazy#compact" do
  it 'returns array without nil elements' do
    arr = [1, nil, 3, false, 5].to_enum.lazy.compact
    arr.should be_an_instance_of(Enumerator::Lazy)
    arr.force.should == [1, 3, false, 5]
  end

  it "sets #size to nil" do
    Enumerator::Lazy.new(Object.new, 100) {}.compact.size.should == nil
  end
end
